# 05-ATM.py
# 查询余额
def get_money(name):
    print('------------------查询余额-----------------')
    print(f'{name},你好，您的余额剩余：{money}')
# 存款
def add_money(name):
    a = int(input('请输入存款金额：'))
    global money  # 提升变量
    money = money+a  
    print(f'{name},你好，你的存款{a}元成功')
    print(f'{name},你好，你的余额剩余{money}元')
# 取款
def update_money(name):
    a = int(input('取款'))
    global money  # 提升变量
    if a<money:
        money = money-a  
        print(f'{name},你好，你的取款{a}元成功')
        print(f'{name},你好，你的余额剩余{money}元')
    else:
        print("余额不足")
# 定义主菜单函数
money = 5000000
def main(name):
    '''
    money:默认银行卡余额
    name:客户的姓名
    '''
    print(f'{name},您好，欢迎来到农商银行ATM,请选择操作:')
    while True:
        print('查询余额','[输入1]')
        print('存款\t','[输入2]')
        print('取款\t','[输入3]')
        print('退出\t','[输入4]')
        index = int(input('请输入您的选择:'))
        if index==4:break
        if index==1:get_money(name)
        if index==2:add_money(name)
        if index==3:update_money(name)
main(input('请输入您的姓名：'))

